from sklearn.naive_bayes import GaussianNB
import numpy as np
from fpdf import FPDF

# 数据集
X = np.array([[20, 5], [22, 3], [25, 7], [30, 2], [35, 1]])  # 特征：年龄和运动频率
y = np.array([1, 1, 1, 0, 0])  # 标签：1 表示喜欢运动，0 表示不喜欢

# 创建模型
model = GaussianNB()

# 训练模型
model.fit(X, y)

# 预测新数据
new_data = np.array([[28, 4]])
prediction = model.predict(new_data)
print("预测结果：", "喜欢运动" if prediction == 1 else "不喜欢运动")